package com.hxx.controller;

import com.hxx.ai.LegalAssiatant;
import com.hxx.ai.LegalPrompt;
import dev.langchain4j.model.input.Prompt;
import dev.langchain4j.model.input.PromptTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.Map;


@RestController
public class LegaController {

    @Autowired
    private LegalAssiatant legalAssiatant;

    @GetMapping("/legal")
    public String legal(String question){
        return legalAssiatant.chat(question);
    }

    @GetMapping("/legal1")
    public String legal1(String question){
        LegalPrompt legalPrompt = new LegalPrompt();
        legalPrompt.setLegal("著作权");
        legalPrompt.setQuestion(question);
        return legalAssiatant.chat(legalPrompt);
    }

    //使用站位符号
    @GetMapping("/chat2")
    public String chat2(String prompt){
        PromptTemplate template = PromptTemplate.from("请解释法律中的'{{it}}'概念");
        Prompt apply = template.apply(prompt);
        return apply.text();
    }

    //使用站位符号
    @GetMapping("/chat3")
    public String chat3(String prompt){
        PromptTemplate template = PromptTemplate.from("请解释法律中的'{{legal1}}'概念");
        Prompt apply = template.apply(Map.of("legal1",prompt));
        return apply.text();
    }
}


























